package com.sun.generator;


import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;
import com.baomidou.mybatisplus.generator.fill.Property;

import java.util.Scanner;

/**
 * @ClassName: CodeGenerator
 * @Description: 代码生成器
 * @Author: SunRui
 * @Date: 2024-12-09 09:42
 * @Version: 1.0
 **/
public class CodeGenerator {
    public static void main(String[] args) {
        // 使用 FastAutoGenerator 快速配置代码生成器
        //数据库配置
        FastAutoGenerator.create("jdbc:mysql://localhost:3306/20250109_category",
                        "root", "1314520sr")
                //全局配置
                .globalConfig(builder -> {
                    builder.author("SunRui") // 设置作者
                            .dateType(DateType.ONLY_DATE)// 设置时间类型策略
                            .commentDate("yyyy-MM-dd") // 设置注释日期格式
                            .enableSpringdoc()
                            .outputDir("src/main/java"); // 输出目录
                })
                //包配置
                .packageConfig(builder -> {
                    builder.parent("com.sun") // 设置父包名
                            .entity("entity") // 设置实体类包名
                            .mapper("mapper") // 设置 Mapper 接口包名
                            .service("service") // 设置 Service 接口包名
                            .serviceImpl("service.impl") // 设置 Service 实现类包名
                            .controller("controller") // 设置 Controller 包名
                            .xml("mappers"); // 设置 Mapper XML 文件包名
                })
                //策略配置
                .strategyConfig(builder -> {
                    builder.addInclude(scanner("表名，多个英文逗号分割").split(",")) // 设置需要生成的表名
                            .entityBuilder()
//                            .superClass(BaseEntity.class)//父类 父类里面可以空着
                            .disableSerialVersionUID()//禁用生成 serialVersionUID
                            .naming(NamingStrategy.underline_to_camel)//数据库表映射到实体的命名策略
                            .columnNaming(NamingStrategy.underline_to_camel)//下划线转驼峰命名
//                            .addSuperEntityColumns("id", "created_by", "created_at", "updated_by", "updated_at")//添加父类公共字段
                            .enableLombok() // 启用 Lombok
                            .enableTableFieldAnnotation() // 启用字段注解
                            .addTableFills(new Column("created_at", FieldFill.INSERT))
                            .addTableFills(new Property("updated_at", FieldFill.INSERT_UPDATE))
                            .idType(IdType.AUTO)
                            .controllerBuilder()
                            .enableRestStyle(); // 启用 REST 风格
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用 Freemarker 模板引擎
                .execute(); // 执行生成

    }

    /**
     * <p>
     * 读取控制台内容
     * </p>
     */
    public static String scanner(String tip) {
        Scanner scanner = new Scanner(System.in);
        StringBuilder help = new StringBuilder();
        help.append("请输入" + tip + "：");
        System.out.println(help.toString());
        if (scanner.hasNext()) {
            String ipt = scanner.next();
            if (StringUtils.isNotEmpty(ipt)) {
                return ipt;
            }
        }
        throw new MybatisPlusException("请输入正确的" + tip + "！");
    }


}